package flash3.net
{
	import flash.utils.ByteArray;
	import flash.utils.getTimer;
	import flash3.net.ByteArrayPlus;
	import flash3.net.MsgPacket;

	public class Encrypt
	{
		public var m_dwSendXorKey:int = 0;
		public var m_dwRecvXorKey:int = 0;
		public var m_cbSendRound:Number = 0;
		public var m_cbRecvRound:Number = 0;
		public var m_dwSendPacketCount = 0;
		
		public static const BYTE_COUNT_32:Number = 4;
		
		public function Encrypt()
		{
		}
		
		//发送字节映射表
		private static const m_SendByteMap:Array =
		[
			0x70,0x2F,0x40,0x5F,0x44,0x8E,0x6E,0x45,0x7E,0xAB,0x2C,0x1F,0xB4,0xAC,0x9D,0x91,
			0x0D,0x36,0x9B,0x0B,0xD4,0xC4,0x39,0x74,0xBF,0x23,0x16,0x14,0x06,0xEB,0x04,0x3E,
			0x12,0x5C,0x8B,0xBC,0x61,0x63,0xF6,0xA5,0xE1,0x65,0xD8,0xF5,0x5A,0x07,0xF0,0x13,
			0xF2,0x20,0x6B,0x4A,0x24,0x59,0x89,0x64,0xD7,0x42,0x6A,0x5E,0x3D,0x0A,0x77,0xE0,
			0x80,0x27,0xB8,0xC5,0x8C,0x0E,0xFA,0x8A,0xD5,0x29,0x56,0x57,0x6C,0x53,0x67,0x41,
			0xE8,0x00,0x1A,0xCE,0x86,0x83,0xB0,0x22,0x28,0x4D,0x3F,0x26,0x46,0x4F,0x6F,0x2B,
			0x72,0x3A,0xF1,0x8D,0x97,0x95,0x49,0x84,0xE5,0xE3,0x79,0x8F,0x51,0x10,0xA8,0x82,
			0xC6,0xDD,0xFF,0xFC,0xE4,0xCF,0xB3,0x09,0x5D,0xEA,0x9C,0x34,0xF9,0x17,0x9F,0xDA,
			0x87,0xF8,0x15,0x05,0x3C,0xD3,0xA4,0x85,0x2E,0xFB,0xEE,0x47,0x3B,0xEF,0x37,0x7F,
			0x93,0xAF,0x69,0x0C,0x71,0x31,0xDE,0x21,0x75,0xA0,0xAA,0xBA,0x7C,0x38,0x02,0xB7,
			0x81,0x01,0xFD,0xE7,0x1D,0xCC,0xCD,0xBD,0x1B,0x7A,0x2A,0xAD,0x66,0xBE,0x55,0x33,
			0x03,0xDB,0x88,0xB2,0x1E,0x4E,0xB9,0xE6,0xC2,0xF7,0xCB,0x7D,0xC9,0x62,0xC3,0xA6,
			0xDC,0xA7,0x50,0xB5,0x4B,0x94,0xC0,0x92,0x4C,0x11,0x5B,0x78,0xD9,0xB1,0xED,0x19,
			0xE9,0xA1,0x1C,0xB6,0x32,0x99,0xA3,0x76,0x9E,0x7B,0x6D,0x9A,0x30,0xD6,0xA9,0x25,
			0xC7,0xAE,0x96,0x35,0xD0,0xBB,0xD2,0xC8,0xA2,0x08,0xF3,0xD1,0x73,0xF4,0x48,0x2D,
			0x90,0xCA,0xE2,0x58,0xC1,0x18,0x52,0xFE,0xDF,0x68,0x98,0x54,0xEC,0x60,0x43,0x0F
		];
		
		//接收字节映射表
		private static const m_RecvByteMap:Array =	
		[
			0x51,0xA1,0x9E,0xB0,0x1E,0x83,0x1C,0x2D,0xE9,0x77,0x3D,0x13,0x93,0x10,0x45,0xFF,
			0x6D,0xC9,0x20,0x2F,0x1B,0x82,0x1A,0x7D,0xF5,0xCF,0x52,0xA8,0xD2,0xA4,0xB4,0x0B,
			0x31,0x97,0x57,0x19,0x34,0xDF,0x5B,0x41,0x58,0x49,0xAA,0x5F,0x0A,0xEF,0x88,0x01,
			0xDC,0x95,0xD4,0xAF,0x7B,0xE3,0x11,0x8E,0x9D,0x16,0x61,0x8C,0x84,0x3C,0x1F,0x5A,
			0x02,0x4F,0x39,0xFE,0x04,0x07,0x5C,0x8B,0xEE,0x66,0x33,0xC4,0xC8,0x59,0xB5,0x5D,
			0xC2,0x6C,0xF6,0x4D,0xFB,0xAE,0x4A,0x4B,0xF3,0x35,0x2C,0xCA,0x21,0x78,0x3B,0x03,
			0xFD,0x24,0xBD,0x25,0x37,0x29,0xAC,0x4E,0xF9,0x92,0x3A,0x32,0x4C,0xDA,0x06,0x5E,
			0x00,0x94,0x60,0xEC,0x17,0x98,0xD7,0x3E,0xCB,0x6A,0xA9,0xD9,0x9C,0xBB,0x08,0x8F,
			0x40,0xA0,0x6F,0x55,0x67,0x87,0x54,0x80,0xB2,0x36,0x47,0x22,0x44,0x63,0x05,0x6B,
			0xF0,0x0F,0xC7,0x90,0xC5,0x65,0xE2,0x64,0xFA,0xD5,0xDB,0x12,0x7A,0x0E,0xD8,0x7E,
			0x99,0xD1,0xE8,0xD6,0x86,0x27,0xBF,0xC1,0x6E,0xDE,0x9A,0x09,0x0D,0xAB,0xE1,0x91,
			0x56,0xCD,0xB3,0x76,0x0C,0xC3,0xD3,0x9F,0x42,0xB6,0x9B,0xE5,0x23,0xA7,0xAD,0x18,
			0xC6,0xF4,0xB8,0xBE,0x15,0x43,0x70,0xE0,0xE7,0xBC,0xF1,0xBA,0xA5,0xA6,0x53,0x75,
			0xE4,0xEB,0xE6,0x85,0x14,0x48,0xDD,0x38,0x2A,0xCC,0x7F,0xB1,0xC0,0x71,0x96,0xF8,
			0x3F,0x28,0xF2,0x69,0x74,0x68,0xB7,0xA3,0x50,0xD0,0x79,0x1D,0xFC,0xCE,0x8A,0x8D,
			0x2E,0x62,0x30,0xEA,0xED,0x2B,0x26,0xB9,0x81,0x7C,0x46,0x89,0x73,0xA2,0xF7,0x72
		];
		
		private static const g_dwPacketKey:Number=0xA55AA55A;
		
		public function Reset():void
		{
			m_cbSendRound = 0;
			m_cbRecvRound = 0;
		}
		
		//随机映射
		public function SeedRandMap(wSeed:int):int
		{
			var dwHold:int = wSeed;
			return (int)((dwHold = dwHold*241103+2533101)>>16);
		}
		
		//映射发送数据
		public function MapSendByte(cbData:int):int
		{
			var cbMap:int=m_SendByteMap[(int)(cbData+m_cbSendRound)];
			m_cbSendRound+=3;
			return cbMap;
		}
			
		//映射接收数据
		public function MapRecvByte(cbData:int):int
		{
			var cbMap:int=m_RecvByteMap[cbData]-m_cbRecvRound;
			m_cbRecvRound+=3;
			return cbMap;
		}
		
		public static function randomArray():Array
		{
			var num_arr:Array = new Array();  
			var j = 0;  
			var judge = 0;  
			do {  
				var temp_num = Math.round(Math.random()*100);  
				for (var i = 0; i<num_arr.length; i++) {  
					var temp_n = num_arr[i];  
					if (temp_num == temp_n) {  
						judge = 1;  
						break;  
					} else {  
						judge = 0;  
					}  
				}  
				if (judge == 0) {  
					num_arr.push(temp_num);  
					j++;  
					_root["t"+j].text = temp_num;  
				}  
			} while (j<100);  
			
			return num_arr;
		}
		
		//加密数据
		public function EncryptBuffer(packet:MsgPacket):ByteArray
		{
			var dataBap:ByteArrayPlus = new ByteArrayPlus();
			var encryptData:ByteArrayPlus = new ByteArrayPlus();
			
			dataBap.writeShort(packet._id);
			dataBap.writeShort(packet._subid);
			dataBap.writeBytes(packet._data);
			
			var checkCode:uint = 0;
			var length:Number = dataBap.length;
			var byte:int = 0;
			for(var i=0;i<length;i++)
			{
				byte = dataBap.readByte();
				checkCode += byte;
				encryptData.writeByte(MapSendByte(byte));
			}
			
			encryptData.position = 0;
			encryptData.writeByte(~checkCode + 1);
			encryptData.writeByte(MsgPacket.VERSION);
			
			//创建密钥
			var dwXorKey:int=m_dwSendXorKey;
			if (m_dwSendPacketCount==0)
			{
				//生成第一次随机种子
				var Guid:Array = randomArray();
				dwXorKey=getTimer()*getTimer();
				dwXorKey^=Guid[0];
				dwXorKey^=Guid[1];
				dwXorKey^=Guid[2];
				dwXorKey^=Guid[3]);
				
				//随机映射种子
				dwXorKey=SeedRandMap((uint)dwXorKey);
				dwXorKey|=((int)SeedRandMap((uint)(dwXorKey>>16)))<<16;
				dwXorKey^=g_dwPacketKey;
				m_dwSendXorKey=dwXorKey;
				m_dwRecvXorKey=dwXorKey;
			}

			var wEncryptSize:uint = _encryptData.length;
			var wSnapCount:uint = 0;
			if ((wEncryptSize % BYTE_COUNT_32)!=0)
			{
				wSnapCount = BYTE_COUNT_32 - wEncryptSize % BYTE_COUNT_32;
				encryptData.length += wSnapCount;
			}
			
			//加密数据
			var pwSeed:int=(int)(pcbDataBuffer+sizeof(CMD_Info));
			var pdwXor:int=(int)(pcbDataBuffer+sizeof(CMD_Info));
			var wEncrypCount:int=(wEncryptSize+wSnapCount) / BYTE_COUNT_32;
			for (i=0;i<wEncrypCount;i++)
			{
				pdwXor++^=dwXorKey;
				dwXorKey=SeedRandMap(*pwSeed++);
				dwXorKey|=((int)SeedRandMap(*pwSeed++))<<16;
				dwXorKey^=g_dwPacketKey;
			}
			
			//插入密钥
			if (m_dwSendPacketCount==0)
			{
				encryptData.position = 8;
				encryptData.writeInt(m_dwSendXorKey);
			}
			
			//设置变量
			m_dwSendPacketCount++;
			m_dwSendXorKey=dwXorKey;
			
			return encryptData;
		}
		
		//解密数据
		public function CrevasseBuffer(data:ByteArrayPlus):MsgPacket
		{
			var wDataSize:uint = data.length;
			var wSnapCount:uint=0;
			if ((wDataSize%BYTE_COUNT_32) != 0)
			{
				wSnapCount = BYTE_COUNT_32 - wDataSize%BYTE_COUNT_32;
				data.length += wSnapCount;
			}
			
			//解密数据
			var dwXorKey:int=m_dwRecvXorKey;
			var pdwXor:int=(int)(pcbDataBuffer+sizeof(CMD_Info));
			var pwSeed:int=(int)(pcbDataBuffer+sizeof(CMD_Info));
			var wEncrypCount:int=(wDataSize+wSnapCount-4)/4;
			data.position = 2;
			for (var i:int=0;i<wEncrypCount;i++)
			{
				if ((i==(wEncrypCount-1))&&(wSnapCount>0))
				{
					var pcbKey:int=((uint)&m_dwRecvXorKey)+BYTE_COUNT_32-wSnapCount;
					data.writeByte(pcbKey);
				}
				dwXorKey=SeedRandMap(data.readShort());
				dwXorKey|=((int)SeedRandMap(data.readShort()))<<16;
				dwXorKey^=g_dwPacketKey;
				*pdwXor++^=m_dwRecvXorKey;
				m_dwRecvXorKey=dwXorKey;
			}
			
			var size:uint = data.readShort();
			var checkCode:uint = data.readByte();
			var version:uint = data.readByte();
			var mainid:uint = data.readShort();
			var subid:uint = data.readShort();
			var byte:uint;
			var encryptData:ByteArrayPlus = new ByteArrayPlus();
			for(var i:int = 0;i<data.length;i++)
			{
				byte = MapRecvByte(data.readByte());
				checkCode += byte;
				encryptData.writeByte(byte);
			}
			//效验码与字节映射
			CMD_Head * pHead=(CMD_Head *)pcbDataBuffer;
			int cbCheckCode=pHead->CmdInfo.cbCheckCode;
			for (int i=sizeof(CMD_Info);i<wDataSize;i++)
			{
				pcbDataBuffer[i]=MapRecvByte(pcbDataBuffer[i]);
				checkCode+=pcbDataBuffer[i];
			}
			if (checkCode!=0) 
				return null;
			
			var packet:MsgPacket = new MsgPacket();
			packet._id = mainid;
			packet._subid = subid;
			packet._data = encryptData;
			
			return packet;
		}
	}
}